package math;

import java.util.Arrays;

/**
 * question: 一个整数数组中，查找最长的子数组，该子数组的元素和不超过给定值
 *
 * @author Liaorun
 */
public class LongestLessSumSubArrayLength {

    public static void main(String[] args) {
        int res = maxLengthAwesome(new int[]{-100, 20, 20, 30, 10, 200}, 70);
        System.out.println("result: " + res);
    }


    public static int maxLengthAwesome(int[] arr, int k) {
        if (arr == null || arr.length == 0) {
            return 0;
        }

        // 记录i位置开始向右，能够得到的最小元素和
        int[] minSums = new int[arr.length];
        // 记录上一个数组对应的结束位置
        int[] minSumEnds = new int[arr.length];



        // 数组的最后一个位置右边没有值，可以直接赋值
        minSums[arr.length - 1] = arr[arr.length - 1];
        minSumEnds[arr.length - 1] = arr.length - 1;

        for (int i = arr.length - 2; i >= 0; i--) {
            if (minSums[i + 1] <= 0) {
                // 如果右边位置记录的最小和为负数 or 0，表示当前位置向右边延展可以获得更小的累加和
                minSums[i] = arr[i] + minSums[i + 1];
                minSumEnds[i] = minSumEnds[i + 1];
            } else {
                // 如果右边位置记录的最小和为正数，表示当前位置向右边延展不能获得更小的累加和
                // 就当前位置一个数为一个窗口时，可以获得最小值
                minSums[i] = arr[i];
                minSumEnds[i] = i;
            }
        }
        System.out.println(Arrays.toString(minSums));
        System.out.println(Arrays.toString(minSumEnds));

        int end = 0;
        int sum = 0;
        int res = 0;

        // i是窗口的最左位置，end是窗口最右的位置的下一个位置（终止位置）
        for (int i = 0; i < arr.length; i++) {
            // 跳出条件：找到i开始的窗口，在 <= k的情况下最大的窗口
            // 是否没越界 && 加上下一块，窗口的和任然 <= k
            while (end < arr.length && sum + minSums[end] <= k) {
                // 更新目前窗口元素和
                sum += minSums[end];
                // 更新窗口结束点
                end = minSumEnds[end] + 1;
            }
            // 当前窗口十分最大，可以更新大小
            res = Math.max(res, end - i);
            if (end > i) {
                // 窗口中还有数，减去窗口中第一个数字，为了下次循环准备
                sum -= arr[i];
            } else {
                // 窗口中没有数，说明从i开头的所有子数组累加和都不可能 <= k
                end = i + 1;
            }
        }

        return res;
    }
}
